import os
import pandas as pd
import pyperclip

# 1. 获取剪贴板中的目录路径
folder_path = pyperclip.paste()

# 2. 检查路径有效性
if not os.path.isdir(folder_path):
    raise ValueError("剪贴板中的路径无效，请确认复制的是一个有效的文件夹路径。")

# 3. 获取指定目录下所有doc、docx、wps文件的文件名
valid_extensions = ['.doc', '.docx', '.wps']
file_names = [f for f in os.listdir(folder_path)
              if os.path.isfile(os.path.join(folder_path, f)) and os.path.splitext(f)[1].lower() in valid_extensions]

# 4. Excel文件路径
excel_path = os.path.join(os.getcwd(), '稿件工作进度表.xlsx')

# 5. 读取原有Excel文件或创建新DataFrame
if os.path.exists(excel_path):
    df = pd.read_excel(excel_path)
else:
    df = pd.DataFrame()

# 6. 找出应该插入的位置（最后一行之后）
start_index = len(df)

# 7. 插入新的文件名到第一列
for i, name in enumerate(file_names):
    df.loc[start_index + i, 0] = name

# 8. 保存Excel文件
df.to_excel(excel_path, index=False)

print("文件名已成功写入Excel文件。")
